package repository

import (
	"admin-system-go/model/do"
	"database/sql"
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
	"log"
	"os"
	"strings"
	"time"
)

var db *gorm.DB

func init() {
	dbName := strings.ToUpper(os.Getenv("DB_NAME"))
	if dbName == "MYSQL" {
		initMysql()
	} else {
		initPostgresSql()
	}
	initDB()
}

func initDB() {
	err := db.AutoMigrate(
		&do.User{},
		&do.UserGroup{},
		&do.Role{},
		&do.UserRole{},
		&do.UserGroupRole{},
		&do.GroupUser{},
		&do.GroupResource{},
		&do.Menu{},
		&do.Permission{},
		&do.Resource{},
		&do.ResourceGroup{},
		&do.ResourceGroupRole{},
		&do.RoleMenu{},
		&do.RolePermission{},
		&do.RoleResource{},
		&do.Token{},
	)
	if err != nil {
		fmt.Println(err)
	}
}

var newLogger = logger.New(
	log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
	logger.Config{
		SlowThreshold:             time.Second, // Slow SQL threshold
		LogLevel:                  logger.Info, // Log level
		IgnoreRecordNotFoundError: true,        // Ignore ErrRecordNotFound error for logger
		ParameterizedQueries:      true,        // Don't include params in the SQL log
		Colorful:                  false,       // Disable color
	},
)

func initMysql() {
	dsn := "root:123456@tcp(127.0.0.1:3306)/admin_system?charset=utf8mb4&parseTime=True&loc=Local"
	_db, err := gorm.Open(mysql.New(mysql.Config{
		DSN:                       dsn,
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置
	}), &gorm.Config{
		// 使用CreateBatchSize 选项初始化 GORM 时，所有的创建& 关联 INSERT 都将遵循该选项
		CreateBatchSize: 1000,
		// 全局模式：执行任何 SQL 时都创建并缓存预编译语句，可以提高后续的调用速度
		PrepareStmt: true,
		// 注意 QueryFields 模式会根据当前 model 的所有字段名称进行 select。
		//QueryFields: true,
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true,
		},
		Logger: newLogger,
	})

	if err != nil {
		fmt.Println(err)
	}
	db = _db
	setPoolConfig()
}

func initPostgresSql() {
	dsn := "host=127.0.0.1 user=postgres password=123456 dbname=admin_system port=5432 sslmode=disable TimeZone=Asia/Shanghai"
	_db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
		// 使用CreateBatchSize 选项初始化 GORM 时，所有的创建& 关联 INSERT 都将遵循该选项
		CreateBatchSize: 1000,
		// 全局模式：执行任何 SQL 时都创建并缓存预编译语句，可以提高后续的调用速度
		PrepareStmt: true,
		// 注意 QueryFields 模式会根据当前 model 的所有字段名称进行 select。
		//QueryFields: true,
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true,
		},
		Logger: newLogger,
	})
	if err != nil {
		panic("Failed to connect to PostgresSql database")
	}
	db = _db
	setPoolConfig()
}

func setPoolConfig() {
	sqlDB, _ := db.DB()
	// SetMaxIdleConns sets the maximum number of connections in the idle connection pool.
	sqlDB.SetMaxIdleConns(10)

	// SetMaxOpenConns sets the maximum number of open connections to the database.
	sqlDB.SetMaxOpenConns(100)

	// SetConnMaxLifetime sets the maximum amount of time a connection may be reused.
	sqlDB.SetConnMaxLifetime(time.Hour)
}

func GetDb() *gorm.DB {
	return db
}

func CloseDb() {
	sqlDb, err := db.DB()
	if err != nil {
		fmt.Println("get db err ", err)
	}
	defer func(sqlDb *sql.DB) {
		err := sqlDb.Close()
		if err != nil {
			fmt.Println(err)
		}
	}(sqlDb)
}
